import shutil

import cv2
import os
import scipy.io as sio

root_path = r'E:\uav_datasets\Drone-Detection-RTB'

'''
改进：
1、去除长宽比极端的个例（镜头切换导致），当遇到极端比例的RTB框时，不进行bbox绘制
2、设计帧长度与clip比例：16 = 1:10 28/220
'''


def vid_filter():
    vid_dirs = os.listdir(root_path)  # ['V_AIRPLANE_012', 'V_BIRD_003', ... ]
    total = 0
    for vid_dir in vid_dirs:
        if vid_dir.startswith("MUTI_"): continue  # 跳过存在多目标的视频
        print('当前处理：', vid_dir)
        total += 1
        src_path = os.path.join(root_path, vid_dir)
        dir_files = os.listdir(src_path)  # 目录下的所有文件，包括mat文件和所有视频帧
        # 删除.mp4文件和clips文件夹
        for dir_file in dir_files:
            if dir_file.endswith(".mp4"):
                # TODO 筛选视频
                pass

    print("共处理了：", total)


def testProcBar():
    import time
    from tqdm import tqdm
    for i in tqdm(range(100), desc="处理中", unit="files", ncols=75, colour='#37B6BD', ):
        time.sleep(0.1)
        """
    「参数：」
    range(100)：指定要遍历的范围，这里是 0 到 99。
    desc="下载中"：指定进度条的描述文本，显示在进度条的前面，这里是 “下载中”。
    unit="files"：指定进度条的单位文本，显示在进度值后面，这里是 “files”。
    ncols=75：指定进度条的总宽度（以字符为单位），这里是 75。
    colour='#37B6BD'：指定进度条的颜色，这里是 “#37B6BD 蓝色  #00ff00 绿色”。
    """


if __name__ == '__main__':
    # 285 - 28 = 257
    # recover()
    testProcBar()
